package com.tanhua.autoconfig.templates;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.tanhua.autoconfig.properties.OssProperties;
import org.springframework.beans.factory.annotation.Autowired;

import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

public class OssTemplate {

    @Autowired
    private OssProperties properties;

    /**
     * 文件上传
     * 返回值：图片的请求地址
     * 参数：
     * 文件的输出流，图片名称(1.jpg)
     */
    public String upload(InputStream is, String filename) {
        filename = new SimpleDateFormat("yyyy/MM/dd").format(new Date())
                + "/" + UUID.randomUUID().toString() + filename.substring(filename.lastIndexOf("."));

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(properties.getEndpoint(),
                properties.getAccessKey(),
                properties.getSecret());

        // 1.依次填写Bucket名称 , 2.存放完整路径。3.输入流
        ossClient.putObject(properties.getBucketName(), filename, is);
        // 关闭OSSClient。
        ossClient.shutdown();
        return properties.getUrl() + "/" + filename;
    }


    //需要补充：删除图片方法（参考阿里云的文档）
    public void delete(String imgUrl) {
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(properties.getEndpoint(), properties.getAccessKey(), properties.getSecret());

        // 删除文件或目录。如果要删除目录，目录必须为空。

        ossClient.deleteObject(properties.getBucketName(), imgUrl);

        // 关闭OSSClient。
        ossClient.shutdown();
    }
}
